# 主要是streamlit的入口文件
import streamlit as st
from streamlit_model.streamlit_layout import api_test_ui, functional_test_ui
import streamlit_model.streamlit_utils.session_utils as ss
from streamlit_model.streamlit_utils.session_utils import init_session

init_session()

@ss.require_login
def main():


    # 初始化页面配置
    st.set_page_config(
        page_title="xiaochentest",
        page_icon="✅",
        layout="wide",
        initial_sidebar_state="expanded"
    )

    # 侧边栏导航
    st.sidebar.write(f"欢迎，{st.session_state['username']}！")
    if st.sidebar.button("注销"):
        ss.logout()
        st.rerun()

    selected_tab = st.sidebar.radio("选择功能", ["功能测试用例", "接口测试用例", "用例调试"])

    # 根据选择的标签渲染内容
    if selected_tab == "功能测试用例":
        functional_test_ui.render()

    elif selected_tab == "接口测试用例":
        api_test_ui.render()


if __name__ == "__main__":
    main()
